import pyautogui
import datetime
import keyboard
import os

def take_screenshot():
    # 文件名：年-月-日_时-分-秒
    current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
    filename = f"{current_time}.png"
    # 文件夹
    save_dir = "../ScnPhoto"
    # 拼接完整路径（目录 + 文件名）
    full_path = os.path.join(save_dir, filename)
    # 全屏截图并保存
    pyautogui.screenshot().save(full_path)

if __name__ == "__main__":
    print("程序已启动：")
    print("按 'P' 键触发全屏截图")
    print("按 'Esc' 键退出程序")
    # 注册P键按下事件（触发截图函数）
    keyboard.on_press_key('p', lambda _: take_screenshot())
    keyboard.wait('esc')# 等待Esc键按下后退出程序
